from dataclasses import dataclass, field
from typing import Dict, List
from ..common.self_attention_simulator import SelfAttentionSimulator
from .simulator import (
    LlamaCoreAttentionSimulator,
)

@dataclass
class LlamaSelfAttentionSimulator(SelfAttentionSimulator):

    core_attention: LlamaCoreAttentionSimulator = None